
###
 # @Author: daniel
 # @Date: 2023-08-30 14:43:12
 # @LastEditTime: 2024-01-09 02:39:42
 # @LastEditors: daniel
 # @Description: 
 # @FilePath: /topictrack-bee/daniel_eval_metrics_ant.sh
 # have a nice day
### 

export PYTHONPATH="$PYTHONPATH:/data/xusc/exp/topictrack-bee:/data/xusc/exp/topictrack-bee:/data/xusc/exp/topictrack-bee:/data/xusc/exp/topictrack-bee/external/YOLOX/"

source /usr/local/miniconda3/etc/profile.d/conda.sh 

conda activate beetrack



exp=antmove_test
dataset=antmove # optional: MOT17, MOT20, DANCE, GMOT, BEE23

#* inference, step1
#* origin 
alpha_gate=0.1;
gate=0.3;



# *================================================================================
CUDA_VISIBLE_DEVICES='2' python3 main.py --exp_name $exp \
  --alpha_gate $alpha_gate --gate $gate \
  --aspect_ratio_thresh 1.6 --w_assoc_emb 0.7 \
  --AARM --TOPIC \
  --dataset $dataset \
  --test_dataset

#*================================================================================
#* step 2
python3 external/TrackEval/scripts/run_mot_challenge.py \
        --SPLIT_TO_EVAL val \
        --METRICS HOTA Identity CLEAR \
        --TRACKERS_TO_EVAL ${exp} \
        --GT_FOLDER results/gt/ \
        --TRACKERS_FOLDER results/trackers/ \
        --BENCHMARK $dataset
      
python3 external/TrackEval/scripts/run_mot_challenge.py \
        --SPLIT_TO_EVAL val \
        --METRICS HOTA Identity CLEAR \
        --TRACKERS_TO_EVAL ${exp}_post \
        --GT_FOLDER results/gt/ \
        --TRACKERS_FOLDER results/trackers/ \
        --BENCHMARK $dataset



python3 demo.py --exp_name $exp --dataset $dataset --test_dataset